package vacina;

import java.util.ArrayList;
import java.util.List;
import javax.swing.table.AbstractTableModel;

/**
 *
 * @author Max Weber
 */
public class VacinaModal extends AbstractTableModel {

    private String[] vacinaColunas = {"Cod.","Nome"};
    List<Vacina> vacina = new ArrayList<Vacina>();

    public VacinaModal(List<Vacina> vacina) {
        this.vacina.clear();
        this.vacina.addAll(vacina);
    }

    @Override
    public int getRowCount() {
        return vacina.size();
    }

    @Override
    public int getColumnCount() {
        return vacinaColunas.length;
    }

    @Override
    public Object getValueAt(int rowIndex, int columnIndex) {
        Vacina vacinas = vacina.get(rowIndex);
        switch (columnIndex) {
            case 0:
                return vacinas.getId_vacina();
            case 1:
                return vacinas.getNome_vacina();
        }
        return null;
    }

    public String getColumnName(int column) {
        switch (column) {
            case 0:
                return vacinaColunas[0];
            case 1:
                return vacinaColunas[1];
        }
        return null;
    }
}